import sys
import os

sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)), ".."))
sys.path.insert(0, os.path.join(os.path.dirname(os.path.abspath(__file__)), "..", ".."))


def get_foil_label_fn(type):
    # print(type)
    if type == 'bladder':
        from FOIL.model_label.bladder_foil_for_vscode import FOIL as FOIL
        from FOIL.model_label.bladder_label_for_vscode import label as LABEL
    elif type == 'bladder4':
        from FOIL.model_label.bladder_foil_four_for_vscode import FOIL as FOIL
        from FOIL.model_label.bladder_label_four_for_vscode import label as LABEL
    elif type == 'medical':
        from FOIL.model_label.medical_foil_for_vscode import FOIL as FOIL
        from FOIL.model_label.medical_label_for_vscode import label as LABEL
    elif type == 'breast':
        from FOIL.model_label.breast_foil_for_vscode import FOIL as FOIL
        from FOIL.model_label.breast_label_for_vscode import label as LABEL
    elif type == 'breast_before':
        from FOIL.model_label.breast_foil_before_for_vscode import FOIL as FOIL
        from FOIL.model_label.breast_label_before_for_vscode import label as LABEL
    else:
        raise Exception('Invalid type')
    return FOIL, LABEL
